package stack;

import java.util.Stack;

/**
 * @Description 155. 最小栈
 * @Author Firenut
 * @Date 2023-03-12 17:13
 */
public class T155_MinStack {
    public static void main(String[] args) {
        MinStack minStack = new MinStack();
        minStack.push(0);
        minStack.push(1);
        minStack.push(0);
        minStack.getMin(); // 0
        minStack.pop(); // [0,1]
        minStack.getMin();// [0]

    }
}


// 本题需要使用O(1)的复杂度来获取最小值，
// 所以想到的是借用一个辅助栈，其栈顶存放最小值即可
class MinStack {

    Stack<Integer> stack;
    Stack<Integer> min_stack;
    public MinStack() {
        stack = new Stack<>();
        min_stack = new Stack<>();
    }

    public void push(int val) {
        stack.push(val);
        if(min_stack.isEmpty()||min_stack.peek()>=val){
            //如果值相同也应该push进去,因为后面pop()的时候如果是最小值，
            // 那么min_stack的栈顶元素也应该一起pop(),主要是保证两者能一一对应
            // 也即stack如果pop()出两个最小值，那么min_stack也应该pop()出两个最小值
            min_stack.push(val);
        }
    }

    public void pop() {
        int val= stack.pop();
        if(min_stack.peek()==val){
            min_stack.pop();
        }
    }

    public int top() {
        return stack.peek();
    }

    public int getMin() {
        return min_stack.peek();
    }
}